from .unet import UNet
from .myunet import VaryUNet
from .cnnBE import CNN_BE
model_names = {
    'VaryUNet' :    VaryUNet,
    'UNet' :    UNet,
    'cnnBE': CNN_BE
    }
